#!/bin/bash
ntopics=$1
niterations=$2

# DTM code: https://github.com/Blei-Lab/dtm

# Run DTM script
command="./dtm_release/dtm/main \
  --ntopics=${ntopics} \
  --mode=fit \
  --rng_seed=0 \
  --initialize_lda=true \
  --corpus_prefix=./data \
  --outname=./model_runs/${ntopics}-topics_${niterations}_iterations_dtm_output \
  --top_chain_var=0.005 \
  --alpha=0.01 \
  --lda_sequence_min_iter=5 \
  --lda_sequence_max_iter=${niterations} \
  --lda_max_em_iter=${niterations}"

eval $command

